import java.util.HashMap;
import java.util.Map;

public class SubarraySum {
    // leetcode 和为k的子数组
    // https://leetcode.cn/problems/subarray-sum-equals-k/description/?envType=study-plan-v2&envId=top-100-liked
    public int subarraySum(int[] nums, int k) {
        // 将遍历之后的元素的和都存放在map中，遍历到第i个元素的时候，看是否存在sum-k的元素
        Map<Integer, Integer> map = new HashMap<>();
        // 避免[0,i]的元素和为k
        map.put(0,1);
        int sum = 0, count = 0;
        for (int n : nums) {
            sum += n;
            count += map.getOrDefault(sum-k, 0);
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }
        return count;
    }
}
